package com.gitee.qdbp.base.utils;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.session.Session;
import org.apache.shiro.session.UnknownSessionException;
import org.apache.shiro.subject.Subject;
import com.gitee.qdbp.able.exception.ServiceException;
import com.gitee.qdbp.able.result.ResultCode;
import com.gitee.qdbp.base.system.model.IAccount;
import com.gitee.qdbp.base.system.model.ICascadeResource;
import com.gitee.qdbp.tools.utils.RandomTools;

/**
 * 用户工具类
 *
 * @version 150612
 */
public class SessionTools {

    public static final String RESOURCES = "LoginUserResources";

    /**
     * 获取登录用户信息, 如果用户不存在则报错
     *
     * @return 登录用户
     */
    public static IAccount getLoginUser() throws ServiceException {
        IAccount user = checkLoginUser();
        if (user != null) {
            return user;
        } else {
            throw new ServiceException(ResultCode.UNAUTHORIZED);
        }
    }

    /**
     * 获取登录用户信息, 如果用户不存在则返回null
     *
     * @return 登录用户
     */
    public static IAccount checkLoginUser() {
        try {
            Subject current = SecurityUtils.getSubject();
            Object principal = current.getPrincipal();
            if (principal instanceof IAccount) {
                return (IAccount) principal;
            } else {
                return null;
            }
        } catch (UnknownSessionException e) {
            return null;
        }
    }

    public static void setUserResources(ICascadeResource<?> root) {
        Session session = SecurityUtils.getSubject().getSession();
        session.setAttribute(RESOURCES, root);
    }

    public static ICascadeResource<?> getUserResources() {
        Session session = SecurityUtils.getSubject().getSession();
        return (ICascadeResource<?>) session.getAttribute(RESOURCES);
    }

    /**
     * 获取 session
     *
     * @return session; session不存在时创建一个;
     */
    public static Session getSession() {
        return SecurityUtils.getSubject().getSession();
    }

    /**
     * 获取 session
     *
     * @param create session不存在时是否创建
     * @return session
     */
    public static Session getSession(boolean create) {
        return SecurityUtils.getSubject().getSession(create);
    }

    /**
     * 获取 session id
     *
     * @return session
     */
    public static String getSessionId() {
        Session session = getSession(false);
        return session == null || session.getId() == null ? null : session.getId().toString();
    }

    /**
     * 获取session属性
     *
     * @param name 属性名称
     * @return 属性值; session不存在时返回null; session中没有该属性时返回null
     */
    @SuppressWarnings("unchecked")
    public static <T> T getAttribute(String name) {
        Session session = getSession(false);
        return session == null ? null : (T) session.getAttribute(name);
    }

    /**
     * 设置session属性
     *
     * @param name 属性名称
     * @param value 属性值
     */
    public static void setAttribute(String name, Object value) {
        Session session = getSession();
        session.setAttribute(name, value);
    }

    /**
     * 设置session属性
     *
     * @param name 属性名称
     */
    public static void removeAttribute(String name) {
        Session session = getSession();
        session.removeAttribute(name);
    }

    /**
     * Session已修改, 触发Shiro框架保存到缓存<br>
     * 直接操作session.attribute不需要调用该方法<br>
     * 如果修改的是session中某个对象的属性就必须调用该方法, 否则分布式Session中对象的属性不会更新<br>
     */
    public static void sessionChanged() {
        Session session = SecurityUtils.getSubject().getSession();
        // 随机修改一个属性, 触发NativeSessionManager的onChange(session)
        session.setAttribute("<<NULL>>", RandomTools.generateNumber(6));
        session.removeAttribute("<<NULL>>");
    }

    /** 是不是开发模式 **/
    public static boolean isDevelopMode() {
        // 通过登录时授予指定权限来判断是否处于开发模式
        return SecurityUtils.getSubject().isPermitted("root:develop");
    }
}
